# -*- coding: UTF-8 -*-

import pandas as pd

from gensim.models import Word2Vec

from gensim.models.word2vec import LineSentence

# read after NLP txt 
df = pd.read('./sanguo_nlp.csv').astype(str)

sentences = df['doc']

line_sent = []

for s in sentences:
    line_sent.append(s.split()) # make list

# seve model
model.save('./word2vec01.model')

